package jsp;

import util.CookieUtil;
import util.StringUtil;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.sql.*;

/**
 * @author 石小俊
 * @date 2023年07月03日 16:00
 * 用户在访问时访问该Servlet,将该Servlet当做某个网站的主页
 * 当用户访问该Servlet的时候,会对当前Cookie中的数据进行检查
 * 判断当前Cookie中是否包含username与password
 * 如果存在
 *  则判断当前的username与password是否正确
 *  如果正确
 *      则直接进入到主页main.jsp
 *  如果不正确
 *      移除当前Cookie中保存的username与password
 *      则进入到登录页面login.jsp,需要用户进行登录
 * 如果不存在
 *   则进入到登录页面login.jsp,需要用户进行登录
 */
@WebServlet("/showMain")
public class ShowMainServlet extends HttpServlet {
    protected void service(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        // 取出Cookie中的数据
        String username = CookieUtil.findByName("username",request);
        String password = CookieUtil.findByName("password",request);

        // 如果Cookie中没有取到对应数据,直接跳转到登录页面
        if(StringUtil.isNull(username) || StringUtil.isNull(password)){
            request.getRequestDispatcher("/jsp/login.jsp").forward(request,response);
            return;
        }

        // 如果Cookie中的数据与数据库的数据一致,则表示可以直接跳转到主页,无需登录
        Connection conn = null;
        PreparedStatement ps = null;
        ResultSet rs = null;
        String url = "jdbc:mysql://127.0.0.1:3306/ums?useUnicode=true&characterEncoding=utf-8";
        try {
            Class.forName("com.mysql.jdbc.Driver");
            conn = DriverManager.getConnection(url,"root","");
            String sql = new StringBuffer()
                    .append(" select id,username,password ")
                    .append(" from t_user ")
                    .append(" where username = ? ")
                    .append(" and password = ? ")
                    .toString();
            ps = conn.prepareStatement(sql);
            ps.setString(1,username);
            ps.setString(2,password);
            rs = ps.executeQuery();
            if(rs.next()){
                // 登录成功之后页面直接跳转到主页
                request.getRequestDispatcher("/jsp/main.jsp").forward(request,response);
                return;
            }
            // 如果用户名密码与数据库的不一致,说明当前保存在Cookie中的数据需要被移除
            CookieUtil.deleteCookie("username",response);
            CookieUtil.deleteCookie("password",response);
            // 登录失败转发到登录页面进行登录
            request.getRequestDispatcher("/jsp/login.jsp").forward(request,response);
        } catch (Exception e) {
            e.printStackTrace();
            request.getRequestDispatcher("/jsp/login.jsp").forward(request,response);
        } finally {
            if(conn != null){
                try {
                    conn.close();
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
            if(ps != null){
                try {
                    ps.close();
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
            if(rs != null){
                try {
                    rs.close();
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
        }
    }
}
